package com.justreading.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;

/**
 * @author LYJ
 * @Description
 * @date 2021 年 10 月 31 日 下午6:20
 */
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {


    //ChannelGroup 是netty提供的一个维护所有channels的一个类，如果我们自己去实现的话一般使用一个hashMap去维护所有的channel
    private static ChannelGroup channels = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
    private static SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    /**
     * 当第一次开始建立链接的时候我们将其加入到channels当中
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channels.writeAndFlush("[ 客 户 端 ]" + channel.remoteAddress() + " 加 入 聊 天 " + simpleDateFormat.format(new
                java.util.Date()) + " \n");
        channels.add(channel);
    }


    /**
     * 第一次建立连接服务端打印
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + " 上线了~");
    }

    /**
     * 第一次建立连接客户端打印
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + " 离线了~");
    }

    /**
     * 开始实现自己的逻辑，将自己的channel进行排除，转发给其他的channel
     *
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        Channel myChannel = ctx.channel();
        channels.forEach((channel -> {
            if (channel == myChannel) {
                myChannel.writeAndFlush("[自己]发送了消息" + msg + "\n");
            } else {
                channel.writeAndFlush("[客户]" + myChannel.remoteAddress() + " 发送了消息" + msg + "\n");
            }
        }));
    }


    /**
     * netty给我们提供的group会帮助我们实现自动的删除该channel，当离线的时候，所以不需要我们进行删除
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channels.writeAndFlush("[客户端]" + channel.remoteAddress() + " 离开了\n");
        System.out.println("channelGroup size" + channels.size());
    }
}
